-
Notifications
You must be signed in to change notification settings - Fork 80
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Add FP16 support #257
base: master
Are you sure you want to change the base?
Add FP16 support #257
Conversation
micahcc
commented
Dec 28, 2024
•
edited
Loading
edited
- Adds support for floating point-16 bit images
- Use the half crate since half precision floats are nightly only
- added several tests of different features, including prediction
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Looks good to me.
Only concern is whether we have permission to distribute the test image. Could you share the source of the image and the license it is covered under?
a8fdc9d
to
77a22d6
Compare
@fintelia I removed the previous image which I had pulled from the internet and switched to images generated with gimp. I relinquish all rights. |
- add predictor for fp16 - add support for reading half (fp16) images - add a pure white test image (generated with gimp) - add predictor tests
166282b
to
2d98a70
Compare
@fintelia how do I merge this? |
It looks like half doesn't build with 1.61, should this be feature flagged then? |
Should be fine to bump the MSRV to 1.70. That'll require changes in Cargo.toml and .github/workflows/rust.yml |
This reverts commit c2cb3d1.